package com.fanco.learn_datastruct_leetcode.designpattern.flyweight;

import java.util.HashMap;

/**
 * @author qiaowang 2022/2/16
 * 描述
 */
public class WebSiteFactory {
    private HashMap<String, ConcreteWebSite> pool = new HashMap<>();

    // 根据网站的类型 返回一个网站。如果没有就创建一个，并放入到池中，并返回
    public WebSite getWebSiteCategory(String type) {
        if (!pool.containsKey(type)) {
            pool.put(type, new ConcreteWebSite(type));
        }

        return (WebSite) pool.get(type);
    }

    public int getWebsiteCount() {
        return pool.size();
    }
}
